import os
from dotenv import load_dotenv

load_dotenv()

class Config:

    # 大模型配置
    LLM_URI = os.getenv("LLM_URI","")
    LLM_API_KEY = os.getenv("LLM_API_KEY", "")
    LLM_MODEL = os.getenv("LLM_MODEL")
    USE_THINKING = os.getenv("USE_THINKING", "false").lower() == "true"
    LLM_TIMEOUT = int(os.getenv("LLM_TIMEOUT", 60))

    # 向量模型
    EMBEDDING_MODEL = os.getenv("EMBEDDING_MODEL")
    EMBEDDING_DIM = int(os.getenv("EMBEDDING_DIM", 768))

    # MCP服务
    MCP_URI = os.getenv("MCP_URI")

    # 高德地图MCP
    GAODE_MCP_URI = os.getenv("GAODE_MCP_URI")

    # 迭代次数
    MAX_ITERATIONS = int(os.getenv("MAX_ITERATIONS", 5))
